package 中等.搜索.回溯;

/**
 * 给定三个字符串 s1、s2、s3，请判断 s3 能不能由 s1 和 s2 交织（交错） 组成。
 * 两个字符串 s 和 t 交织 的定义与过程如下，其中每个字符串都会被分割成若干 非空 子字符串：
 * <p>
 * s = s1 + s2 + ... + sn
 * t = t1 + t2 + ... + tm
 * |n - m| <= 1
 * 交织 是 s1 + t1 + s2 + t2 + s3 + t3 + ... 或者 t1 + s1 + t2 + s2 + t3 + s3 + ...
 * 提示：a + b 意味着字符串 a 和 b 连接。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/IY6buf
 */
public class 字符串交织_offer092 {

    public static void main(String[] args) {

        System.out.println(new 字符串交织_offer092().isInterleave(
                "abababababababababababababababababababababababababababababababababababababababababababababababababbb",
                "babababababababababababababababababababababababababababababababababababababababababababababababaaaba",
                "abababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababababbb"));

    }

    private char[] arr1;
    private char[] arr2;
    private char[] arr3;

    /**
     * 双指针+回溯+记忆化
     * 注意：当 s1[i] == s3[r] 或者 s2[j] == s3[r] 时
     * 有两种选择情况
     */
    public boolean isInterleave(String s1, String s2, String s3) {
        if (s1.length() + s2.length() != s3.length()) return false;
        arr1 = s1.toCharArray();
        arr2 = s2.toCharArray();
        arr3 = s3.toCharArray();

        return dfs(0, 0, 0, new boolean[s1.length()][s2.length()]);
    }

    // 表示s1中 [idx1,len1-1] 和 s2中 [idx2,len2-1] 的字符串是否可以构成
    // s3中 [idx3,len3-1]
    private boolean dfs(int idx1, int idx2, int idx3, boolean[][] record) {
        // 访问过，不再访问
        if (idx1 < arr1.length && idx2 < arr2.length &&
                record[idx1][idx2]) return false;

        if (idx1 == arr1.length && idx2 == arr2.length &&
                idx3 == arr3.length) {
            return true;
        }
        if (idx3 == arr3.length) return false;

        boolean ans = false;
        if (idx1 < arr1.length && arr1[idx1] == arr3[idx3]) {
            // 短路运算符成功直接返回
            ans = ans || dfs(idx1 + 1, idx2, idx3 + 1, record);
        }
        if (idx2 < arr2.length && arr2[idx2] == arr3[idx3]) {
            ans = ans || dfs(idx1, idx2 + 1, idx3 + 1, record);
        }

        // 记忆化
        if (idx1 < arr1.length && idx2 < arr2.length) {
            record[idx1][idx2] = true;
        }
        return ans;
    }

}
